# -*- coding: utf-8 -*-
# @Time    : 2021/12/17 1:56 下午
# @Author  : zlt
# @FileName: util_txt.py
# @Software: PyCharm

"""
文件说明：
提取txt文件中的内容加以处理
"""



import os,re

def read_flie(file_name):
    if os.path.exists(file_name) == False:
        return None
    else:
        file_data = ""
        with open(file_name,"r",encoding="utf-8") as f:
            for line in f:
                file_data += line
        return file_data





def alter_file(file_name,new_str,pattern):
    """
    替换文件中需要被替换的内容
    :param file_name: 文件地址
    :param old_str: 旧内容
    :param new_str: 需要被替换的内容
    :return:None
    """
    #如果文件不存在，即返回None
    if os.path.exists(file_name) == False:
        return None
    else:
        #创建要获取文件内容的对象file_data
        file_data = ""
        #开始遍历文件内容，并替换要替换的内容
        with open(file_name,"r",encoding="utf-8") as f:
            for line in f:
                if re.match(pattern,line):
                    line = line.replace(line,new_str)
                file_data += line
        #讲更新后的内容替换到文件中
        with open(file_name,"w",encoding="utf-8") as f:
            f.write(file_data)
        f.close()







